package edu.arizona.arid.akshen.util;

import java.io.ByteArrayOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Enumeration;
import java.util.Hashtable;

import javax.microedition.io.Connector;
import javax.microedition.io.HttpConnection;

/**
 * Source: http://wiki.forum.nokia.com/index.php/HTTP_Post_multipart_file_upload_with_J2ME
 * 
 */
public class HttpMultipartRequest {
	public static final int bufSize = 4096;
	static final String BOUNDARY = "----------V2ymHFg03ehbqgZCaKO6jy";

	byte[] postBytes = null;
	String url = null;

	public HttpMultipartRequest(String url, Hashtable params, String fileField,
			String fileName, String fileType, InputStream in)
			throws Exception {
		this.url = url;

		String boundary = getBoundaryString();

		String boundaryMessage = getBoundaryMessage(boundary, params,
				fileField, fileName, fileType);

		String endBoundary = "\r\n--" + boundary + "--\r\n";

		ByteArrayOutputStream bos = new ByteArrayOutputStream();

		bos.write(boundaryMessage.getBytes());

		int read = 0;
		byte[] buf = new byte[bufSize];
		while((read = in.read(buf)) != -1)
			bos.write(buf, 0, read);

		bos.write(endBoundary.getBytes());

		this.postBytes = bos.toByteArray();

		bos.close();
	}

	String getBoundaryString() {
		return BOUNDARY;
	}

	String getBoundaryMessage(String boundary, Hashtable params,
			String fileField, String fileName, String fileType) {
		StringBuffer res = new StringBuffer("--").append(boundary).append(
				"\r\n");

		Enumeration keys = params.keys();

		while (keys.hasMoreElements()) {
			String key = (String) keys.nextElement();
			String value = (String) params.get(key);

			res.append("Content-Disposition: form-data; name=\"").append(key)
					.append("\"\r\n").append("\r\n").append(value).append(
							"\r\n").append("--").append(boundary)
					.append("\r\n");
		}
		res.append("Content-Disposition: form-data; name=\"").append(fileField)
				.append("\"; filename=\"").append(fileName).append("\"\r\n")
				.append("Content-Type: ").append(fileType).append("\r\n\r\n");

		return res.toString();
	}

	public byte[] send() throws Exception {
		HttpConnection hc = null;

		InputStream is = null;

		ByteArrayOutputStream bos = new ByteArrayOutputStream();

		byte[] res = null;

		try {
			hc = (HttpConnection) Connector.open(url);

			hc.setRequestProperty("Content-Type",
					"multipart/form-data; boundary=" + getBoundaryString());

			hc.setRequestMethod(HttpConnection.POST);

			OutputStream dout = hc.openOutputStream();

			dout.write(postBytes);

			dout.close();

			int ch;

			is = hc.openInputStream();

			while ((ch = is.read()) != -1) {
				bos.write(ch);
			}
			res = bos.toByteArray();
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				if (bos != null)
					bos.close();

				if (is != null)
					is.close();

				if (hc != null)
					hc.close();
			} catch (Exception e2) {
				e2.printStackTrace();
			}
		}
		return res;
	}
}